#pragma once
#include <iostream>
#include <string>
#include <cstring>
#include <functional>
#include <pthread.h>

namespace ThreadModule
{
    static uint32_t number = 1;

    template <typename T>
    class Thread
    {
        using func_t = std::function<void(T)>;

    private:
        void EnableDetach()
        {
            std::cout << "thread's detach flag become to true" << std::endl;
            _isdetach = true;
        }

        void EnableRunning()
        {
            std::cout << "thread is running" << std::endl;
            _isrunning = true;
        }

        static void *Routine(void *args)
        {
            Thread<T> *self = static_cast<Thread<T> *>(args);
            // 将运行标志位置为true
            self->EnableRunning();
            // 如果分离标志位为true，则分离线程
            if (self->_isdetach)
            {
                int n = pthread_detach(self->_tid);
                std::cout << "thread is detached in Routine, the return value is " << n << std::endl;
            }
            pthread_setname_np(self->_tid, self->_name.c_str());
            self->_func(self->_data);

            return nullptr;
        }

    public:
        // 构造函数，需要传入一个入口函数地址
        Thread(func_t func, T data)
            : _tid(0),
              _isdetach(false),
              _isrunning(false),
              res(nullptr),
              _func(func),
              _data(data)
        {
            _name = "thread-" + std::to_string(number++);
        }

        bool Start()
        {
            // 1. 如果线程已经运行起来，防止再次启动，直接返回false
            if (_isrunning)
                return false;

            // 2. 如果线程第一次启动，则创建线程
            // 这里如果Routine不是静态成员函数，默认会有一个this指针参数，与pthread_create中的参数不匹配
            // 所以这里使用静态成员函数，将该线程对象以参数this的形式传给pthread_create
            int n = pthread_create(&_tid, nullptr, Routine, this);
            // 创建线程失败返回false
            if (n != 0)
            {
                std::cerr << "create thread error" << strerror(n) << std::endl;
                return false;
            }
            else
            {
                std::cout << _name << " create success" << std::endl;
                return true;
            }
        }

        void Detach()
        {
            // // 需要处理两种情况
            // // 情况1：在线程还没有启动的时候，调用Detach设置线程分离标志位，然后线程启动之后在Routine函数中进行分离
            // // 情况2：在线程启动之后调用Detach设置线程分离标志位，以及分离线程
            // 如果线程已经分离，直接返回
            if (_isdetach)
            {
                std::cout << _name << " is already detached. No further action needed." << std::endl;
                return;
            }

            // 如果线程还没有启动，设置线程分离标志位
            if (!_isrunning)
            {
                EnableDetach();
                return;
            }
            else
            {
                // 启动后设置线程分离，需要设置标志位之后再进行线程分离
                EnableDetach();
                int n = pthread_detach(_tid);
                std::cout << "thread is detched, the return value is " << n << std::endl;
            }
        }

        bool Stop()
        {
            // 如果运行标志位为true，取消线程并将运行标志位置为false
            if (_isrunning)
            {
                int n = pthread_cancel(_tid);
                if (n != 0)
                {
                    std::cerr << "cancel thread error" << strerror(n) << std::endl;
                    return false;
                }
                else
                {
                    _isrunning = false;
                    std::cout << _name << " stop" << std::endl;
                    return true;
                }
            }

            return false;
        }

        void Join()
        {
            // 分离的线程不能被等待
            if (_isdetach)
            {
                std::cout << "thread is detached. it can't be joined! " << std::endl;
                return;
            }
            int n = pthread_join(_tid, &res);
            if (n != 0)
            {
                std::cerr << "join thread error" << std::endl;
            }
            else
            {
                std::cout << "join thread success" << std::endl;
            }
        }
        ~Thread()
        {
        }

    private:
        pthread_t _tid;    // 线程ID
        std::string _name; // 线程名字
        bool _isdetach;    // 线程分离标志位
        bool _isrunning;   // 线程运行标志位
        void *res;         // 线程返回值
        func_t _func;      // 线程入口函数
        T _data;           // 任务函数的参数
    };
}
